# 目标：水源涵养_商洛市_Alberts.tif

import os

# 定义变量类型列表
vartype = [
    '水源涵养', '土壤保持', '洪水调蓄',
    '空气净化', '水质净化', '固碳价值',
    '释氧价值', '气候调节', '负氧离子',
]

# 定义一个函数来重命名目录下的.tif文件
def rename_tif_files(directory_path):
    # 获取目录的名字
    dir_name = os.path.basename(os.path.normpath(directory_path))

    # 确保目录路径存在
    if not os.path.exists(directory_path):
        print(f"The directory {directory_path} does not exist.")
        return

    # 遍历目录中的所有文件
    for filename in os.listdir(directory_path):
        # 检查文件是否是.tif文件
        if filename.lower().endswith('.tif'):
            # 移除文件的扩展名
            base_name = os.path.splitext(filename)[0]
            # 构建新的文件名
            new_filename = f"{dir_name}_{base_name}_Alberts.tif"
            # 获取完整的旧文件路径和新文件路径
            old_file_path = os.path.join(directory_path, filename)
            new_file_path = os.path.join(directory_path, new_filename)
            # 重命名文件
            os.rename(old_file_path, new_file_path)
            print(f"Renamed '{filename}' to '{new_filename}'")

# 遍历vartype列表，为每个值设置directory_path并执行重命名程序
for vt in vartype:
    directory_path = f'./make-town-tif/{vt}'
    rename_tif_files(directory_path)
